'''
Date: 2022-05-11 16:57:36
LastEditTime: 2022-12-08 16:33:52
'''
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# Copyright (c) Megvii, Inc. and its affiliates.
import os

from yolox.exp import Exp as MyExp


class Exp(MyExp):
    def __init__(self):
        super(Exp, self).__init__()
        self.depth = 0.33
        self.width = 0.50
        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]

        # Define yourself dataset path #要改
        self.data_dir = "datasets/coco"
        self.train_ann = "instances_train2017.json"
        self.val_ann = "instances_val2017.json"

        self.num_classes = 4

        self.data_num_workers = 6  #dataloader的数量，一般取cup内核数
    
        ################
        self.max_epoch = 200
        self.eval_interval = 5    #间隔多少个epoch进行一次评估
        self.print_interval = 20  # 输出消息间隔。要小于iter数，iter=图片数/betch size。 如果一直没有输出就调小直到1
        ###############

        # self.no_aug_epochs = 10  # 训练结尾，不使用马赛克的epoch数